﻿using System;
using System.Text.RegularExpressions;
using System.Text;
using Microsoft.AspNetCore.Http;

namespace RxxAdmin.Common.Extensions;

public static class HttpContextExtension
{
	public static ISession GetSession(this HttpContext context)
	{
		try
		{
			return context.Session;
		}
		catch (Exception)
		{
			return default;
		}
	}
    /// <summary>
    /// 设置文件下载名称
    /// </summary>
    /// <param name="httpContext"></param>
    /// <param name="fileName"></param>
    public static void FileInlineHandle(this HttpContext httpContext, string fileName)
    {
        string encodeFilename = System.Web.HttpUtility.UrlEncode(fileName, Encoding.GetEncoding("UTF-8"));
        httpContext.Response.Headers.Add("Content-Disposition", "inline;filename=" + encodeFilename);

    }

    /// <summary>
    /// 设置文件附件名称
    /// </summary>
    /// <param name="httpContext"></param>
    /// <param name="fileName"></param>
    public static void FileAttachmentHandle(this HttpContext httpContext, string fileName)
    {
        string encodeFilename = System.Web.HttpUtility.UrlEncode(fileName, Encoding.GetEncoding("UTF-8"));
        httpContext.Response.Headers.Add("Content-Disposition", "attachment;filename=" + encodeFilename);

    }

    /// <summary>
    /// 获取语言种类
    /// </summary>
    /// <param name="httpContext"></param>
    /// <returns></returns>
    public static string GetLanguage(this HttpContext httpContext)
    {
        string res = "zh-CN";
        var str = httpContext.Request.Headers["Accept-Language"].FirstOrDefault();
        if (str is not null)
        {
            res = str.Split(",")[0];
        }
        return res;

    }

    /// <summary>
    /// 判断是否为异步请求
    /// </summary>
    /// <param name="request"></param>
    /// <returns></returns>
    public static bool IsAjaxRequest(this HttpRequest request)
    {
        string header = request.Headers["X-Requested-With"];
        return "XMLHttpRequest".Equals(header);
    }
    /// <summary>
    /// 获取客户端IP
    /// </summary>
    /// <param name="context"></param>
    /// <returns></returns>
    public static string GetClientIp(this HttpContext context)
    {
        if (context == null) return "";
        var result = context.Request.Headers["X-Forwarded-For"].FirstOrDefault();
        if (string.IsNullOrEmpty(result))
        {
            result = context.Connection.RemoteIpAddress?.ToString();
        }
        if (string.IsNullOrEmpty(result) || result.Contains("::1"))
            result = "127.0.0.1";

        result = result.Replace("::ffff:", "127.0.0.1");
        //如果有端口号，删除端口号
        result = Regex.Replace(result, @":\d{1,5}$", "");
        //Ip规则校验
        var regResult =
            Regex.IsMatch(result, @"^((2[0-4]\d|25[0-5]|[01]?\d\d?)\.){3}(2[0-4]\d|25[0-5]|[01]?\d\d?)$")
            || Regex.IsMatch(result, @"^((2[0-4]\d|25[0-5]|[01]?\d\d?)\.){3}(2[0-4]\d|25[0-5]|[01]?\d\d?):\d{1,5}$");

        result = regResult ? result : "127.0.0.1";
        return result;
    }

    /// <summary>
    /// 获取浏览器标识
    /// </summary>
    /// <param name="context"></param>
    /// <returns></returns>
    public static string GetUserAgent(this HttpContext context)
    {
        return context.Request.Headers["User-Agent"];
    }

    public static string[]? GetUserPermissions(this HttpContext context, string permissionsName)
    {
        return context.User.Claims.Where(x => x.Type == permissionsName).Select(x => x.Value).ToArray();
    }

    /// <summary>
    /// 判断是否是 WebSocket 请求
    /// </summary>
    /// <param name="context"></param>
    /// <returns></returns>
    public static bool IsWebSocketRequest(this HttpContext context)
    {
        return context.WebSockets.IsWebSocketRequest || context.Request.Path == "/ws";
    }
}